tensor([[[[[0.2800, 0.0000],
           [0.4900, 0.2300]],

          [[0.5500, 0.2700],
           [0.2700, 0.0000]]],


         [[[0.5500, 0.2700],
           [0.2700, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.5100]],

          [[0.0000, 0.3400],
           [0.3400, 0.5900]]],


         [[[0.0000, 0.3400],
           [0.3400, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.4200, 1.0000],
           [0.0000, 0.0000]],

          [[0.4500, 0.1200],
           [0.1200, 0.4100]]],


         [[[0.4500, 0.1200],
           [0.1200, 0.4400]],

          [[1.0000, 1.0000],
           [1.0000, 1.0000]]]],



        [[[[0.3100, 0.0000],
           [0.5100, 0.2600]],

          [[0.0000, 0.2700],
           [0.2700, 0.0000]]],


         [[[0.0000, 0.2700],
           [0.2700, 0.5600]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]]], grad_fn=<RoundBackward1>)